Pytorch: RuntimeError: zero-dimensional tensor (at position 0) cannot be concatenated